# -*- coding: utf-8 -*-
#强度分层
#伪彩图分层
from skimage import data,color
import matplotlib.pyplot as plt
import numpy as np

def wct(img):
    grayimg=color.rgb2gray(img)#将彩色图像转换为灰度图像
    rows,cols=grayimg.shape
    labels=np.zeros([rows,cols])
    for i in range(rows):
        for j in range(cols):
            if (grayimg[i,j])<0.4:
                labels[i,j]=0
            elif(grayimg[i,j]<0.8):
                labels[i,j]=1
            else:
                labels[i,j]=2
    psdimg=color.label2rgb(labels)#不同的灰度区间采用不同的颜色
    return psdimg

#检验调用上面的函数
#img=data.coffee()
#from PIL import Image
#img=Image.open('dj_a.jpg')
#a=np.array(img)
#psdimg=wct(a)
##plt.figure()
##plt.axis('off')
#plt.imshow(psdimg)#显示强度分层图像

    